package com.jf.generator;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.c3p0.C3p0Plugin;
import com.jfinal.plugin.druid.DruidPlugin;

import javax.sql.DataSource;

public class GenerateModel {

    public static DataSource getDataSource() {
        Prop p = PropKit.use("db.properties");
        DruidPlugin druidPlugin = new DruidPlugin(p.get("jdbcUrl"), p.get("user"), p.get("password"));
        druidPlugin.start();
        return druidPlugin.getDataSource();
    }


    public static void main(String[] args) {
        // base model 所使用的包名
        String baseModelPkg = "com.jf.model.base";
        // base model 文件保存路径
        String baseModelDir = PathKit.getWebRootPath() + "/src/main/java/com/jf/model/base";
//        String baseModelDir = PathKit.getWebRootPath() + "/../src/com/jf/model/base";

    // model 所使用的包名
        String modelPkg = "com.jf.model";
    // model 文件保存路径
        String modelDir = baseModelDir + "/..";

        Generator gernerator = new Generator(getDataSource(), baseModelPkg, baseModelDir, modelPkg, modelDir);
        // 添加不需要生成的表名
        gernerator.addExcludedTable("adv");
        // 设置是否在 Model 中生成 dao 对象
        gernerator.setGenerateDaoInModel(true);
        // 设置是否生成字典文件
        gernerator.setGenerateDataDictionary(true);
        // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
        gernerator.setRemovedTableNamePrefixes("t_","g_");
        gernerator.generate();
    }
}
